﻿<%@ WebHandler Language="C#" Class="Upload" %>

/**
 * KindEditor ASP.NET
 *
 * 本ASP.NET程序是演示程序，建议不要直接在实际项目中使用。
 * 如果您确定直接使用本程序，使用之前请仔细确认相关安全设置。
 *
 */

using System;
using System.Collections;
using System.Web;
using System.IO;
using System.Globalization;
using LitJson;

public class Upload : IHttpHandler
{
    private HttpContext context;

    public void ProcessRequest(HttpContext context)
    {
        //定义允许上传的文件扩展名
        Hashtable extTable = new Hashtable();
        extTable.Add("image", "gif,jpg,jpeg,png,bmp");
        extTable.Add("flash", "swf,flv");
        extTable.Add("media", "swf,flv,mp3,wav,wma,wmv,mid,avi,mpg,asf,rm,rmvb");
        extTable.Add("file", "doc,docx,xls,xlsx,ppt,htm,html,txt,zip,rar,gz,bz2");
        //最大文件大小
        int maxSize = 1000000;
        this.context = context;

        HttpPostedFile imgFile = context.Request.Files["imgFile"];
        if (imgFile == null)
        {
            showError("请选择文件。");
        }
        //文件类型
        String dirName = context.Request.QueryString["dir"];
        if (String.IsNullOrEmpty(dirName))
        {
            dirName = "image";
        }
        if (!extTable.ContainsKey(dirName))
        {
            showError("目录名不正确。");
        }
        String fileName = imgFile.FileName;
        String fileExt = Path.GetExtension(fileName).ToLower();

        if (imgFile.InputStream == null || imgFile.InputStream.Length > maxSize)
        {
            showError("上传文件大小超过限制。");
        }

        if (String.IsNullOrEmpty(fileExt) || Array.IndexOf(((String)extTable[dirName]).Split(','), fileExt.Substring(1).ToLower()) == -1)
        {
            showError("上传文件扩展名是不允许的扩展名。\n只允许" + ((String)extTable[dirName]) + "格式。");
        }
        var saveDir = context.Request["saveDir"];
        string saveDirStr = null;
        if (saveDir == null)
        {
            saveDirStr = "tmp";
        }
        else
        {
            saveDirStr = saveDir.ToString();
        }
        //是否分布式上传
        bool isDFS = false;
        var dfs = context.Request["dfs"];
        if (dfs != null && dfs.ToString() == "1")
        {
            isDFS = true;
        }
        Hashtable hash = new Hashtable();
        if (isDFS)
        {
            var imageUrl = Cloud.Utility.ConfigHelper.Images_SITE_URL;
            var client = new RestSharp.RestClient(imageUrl);//TODO api基路径
            var request = new RestSharp.RestRequest("file/upload", RestSharp.Method.POST);
            request.AlwaysMultipartFormData = true;
            request.AddHeader("dir", dirName + "/" + saveDirStr);
            request.AddHeader("imageUrl", imageUrl);
            request.AddFile("file", Cloud.Utility.StreamHelper.ConvertToBytes(imgFile.InputStream), context.Server.HtmlEncode(imgFile.FileName));

            var posttask = client.ExecutePostTaskAsync(request);
            if (posttask.Result.StatusCode == System.Net.HttpStatusCode.OK)
            {
                var url = posttask.Result.Content.Trim(new char[] { '[', ']', '"' });
                hash["url"] = url;
            }
            else
            {
                showError("分布式上传失败");
            }
        }
        else
        {
            //文件保存目录
            string savePath = "/upload/kindeditor/" + dirName + "/" + saveDirStr;
            String dirPath = context.Server.MapPath(savePath);
            if (!Directory.Exists(dirPath))
            {
                Directory.CreateDirectory(dirPath);
            }

            String newFileName = DateTime.Now.ToString("_yyyyMMddHHmmss_ffff", DateTimeFormatInfo.InvariantInfo) + fileExt;
            String filePath = dirPath + "/" + newFileName;
            imgFile.SaveAs(filePath);
            hash["url"] = savePath + "/" + newFileName;
        }
        hash["error"] = 0;
        context.Response.AddHeader("Content-Type", "text/html; charset=UTF-8");
        context.Response.Write(JsonMapper.ToJson(hash));
        context.Response.End();
    }

    private void showError(string message)
    {
        Hashtable hash = new Hashtable();
        hash["error"] = 1;
        hash["message"] = message;
        context.Response.AddHeader("Content-Type", "text/html; charset=UTF-8");
        context.Response.Write(JsonMapper.ToJson(hash));
        context.Response.End();
    }

    public bool IsReusable
    {
        get
        {
            return true;
        }
    }
}
